//
// Created by rm on 2021/12/11.
//

#ifndef ARMORDETECTOR12_11_ARMORPLATE_H
#define ARMORDETECTOR12_11_ARMORPLATE_H

#endif //ARMORDETECTOR12_11_ARMORPLATE_H
#pragma once

#define BLUETEAM 0
#define REDTEAM 1

#pragma once
#include <iostream>
#include <vector>
#include <math.h>
#include <opencv2/opencv.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/video.hpp>

class ArmorPlate
{
public:
    cv::Mat pre_image;
    cv::Mat armor_image;
    cv::VideoCapture capture_plate;

    int myteam;


    ArmorPlate();
    bool CameraInit(int device);
    void AutoShoot();
    void ImportVideo();

private:
    void ImgPreprosses(const cv::Mat& src, const cv::Mat& dst);


};


class ArmorRect
{
public:
    cv::RotatedRect armors;
};